import torch
from lstm_model import lstm_model

# 加载预训练模型
model=torch.load("../models/model1.pth")

# 微调模型
model.linear.out_features=3000
print(model)

